# write a pytorch demo
import torch

torch.manual_seed(1)
print(torch.rand(5, 3))

if __name__ == '__main__':
    print('hello world')
